#ifndef __W_UTIL_H_
#define __W_UTIL_H_


#include <iostream>
#include <string>
#include <vector>
#include <memory>
#include <sstream>
#include <fstream>
#include <mysql/mysql.h>
#include <jsoncpp/json/json.h>
#include "logger.hpp"

#include <websocketpp/server.hpp>
#include <websocketpp/config/asio_no_tls.hpp>
typedef websocketpp::server<websocketpp::config::asio> wsserver_t;

using namespace std;

class mysql_util
{
public:
    static MYSQL* mysql_create(const string& host,const string& username,const string&password,
                                const string& dbname,uint16_t port = 3306)
    {
        // 1. 初识化mysql句柄
        MYSQL* mysql = mysql_init(NULL);
        if(mysql == NULL)
        {
            ELOG("mysql init failed");
            return NULL;
        }
        // 2. 连接mysql服务器
        // 参数说明:1 mysql--初始化完成的句柄，2host---连接的mysql服务器的地址，3user---连接的服务器的⽤⼾名
        // 4passwd-连接的服务器的密码，5db ----默认选择的数据库名称，6port---连接的服务器的端⼝： 默认0是3306端⼝
        // 7unix_socket---通信管道⽂件或者socket⽂件，通常置NULL，8client_flag---客⼾端标志位，通常置0
        if(mysql_real_connect(mysql,host.c_str(),username.c_str(),password.c_str(),dbname.c_str(),port, NULL, 0) == NULL)
        {
            ELOG("connect mysql server failed: %s",mysql_error(mysql));
            mysql_close(mysql);
            return NULL;
        }
        // 3. 设置当前客户端的字符集
        if(mysql_set_character_set(mysql,"utf8") != 0)
        {
            ELOG("set client character failed: %s",mysql_error(mysql));
            mysql_close(mysql);
            return NULL;
        }
        return mysql;
    }
    // 执行sql语句
    static bool mysql_exec(MYSQL* mysql,const string& sql)
    {
        int ret = mysql_query(mysql,sql.c_str());
        if(ret != 0)
        {
            ELOG("%s",sql.c_str());
            ELOG("mysql query failed: %s",mysql_error(mysql));
            
            return false;
        }
        return true;
    }
    // 关闭连接，销毁句柄
    static void mysql_destroy(MYSQL* mysql)
    {
        if(mysql != NULL)
        {
            mysql_close(mysql);
        }
        return;
    }
};

class json_util
{
public:
    // 序列化
    static bool serialize(const Json::Value& root, string& str)
    {
        Json::StreamWriterBuilder swb;
        unique_ptr<Json::StreamWriter> sw(swb.newStreamWriter());
        std::stringstream ss;
        int ret = sw->write(root,&ss);
        if(ret != 0)
        {
            ELOG("json serialize failed!");
            return false;
        }
        str = ss.str();
 
        return true;
    }
    // 反序列化
    static bool unserialize(const string& str, Json::Value& root)
    {
        Json::CharReaderBuilder crb;
        unique_ptr<Json::CharReader> cr(crb.newCharReader());

        std::string err;
        bool ret = cr->parse(str.c_str(), str.c_str() + str.size(), &root, &err);
        if(ret == false)
        {
            ELOG("json unserialize failed: %s",err.c_str()) ;
            return false;
        }
        return true;
    }
};

class string_util
{
public:
    static int split(const string& src, const string& sep, vector<string>& res)
    {
        size_t pos, idx = 0;

        while(idx < src.size())
        {
            pos = src.find(sep,idx);
            if(pos == string::npos) //找不到分隔符，说明可能是最后一个子串了，直接插入
            {
                res.push_back(src.substr(idx));
                break;
            }
            if(pos == idx)  //出现连续的分隔符，要跳过
            {
                idx += sep.size();
                continue;
            }
            res.push_back(src.substr(idx, pos-idx));
            idx = pos + sep.size();
        }
        return res.size();

    }
};

class file_util
{
public:
    static bool read(const string& filename, string& body)
    {
        // 1. 打开文件
        ifstream ifs(filename,ios::binary);
        if(ifs.is_open() == false)
        {
            ELOG("%s file open failed!",filename.c_str());
            return false;
        }
        // 2. 获取文件大小
        size_t fsize = 0;
        ifs.seekg(0, ios::end);
        fsize = ifs.tellg();
        ifs.seekg(0, ios::beg);
        body.resize(fsize);
        // 3. 读取文件内容
        ifs.read(&body[0], fsize);
        if(ifs.good() == false)
        {
            ELOG("read %s file connet failed!",filename);
            ifs.close();
            return false;
        }

        // 4. 关闭文件
        ifs.close();
        return true;
    }
};
#endif 